from fastapi.responses import HTMLResponse, FileResponse, PlainTextResponse
from pathlib import Path
from fastapi.staticfiles import StaticFiles
from multimedia_tools.schemas import InputMessage, ReturnMessage


async def create_pic(input: InputMessage) -> ReturnMessage:
    output = ReturnMessage(
        prompt=input.prompt,  # 直接访问属性而不是get方法
        url="http://localhost:9201/pic/",  # 处理None值
        task_id=input.style,  # 示例task_id
        # style=input.style,  # 直接访问属性而不是get方法
    )
    return output
    # return {
    #     "prompt": input.prompt,  # 直接访问属性而不是get方法
    #     "url": "http://localhost:9201/pic/",  # 处理None值
    #     "task_id": "12345",  # 示例task_id
    #     "style": input.style,  # 直接访问属性而不是get方法
    # }


async def create_video() -> PlainTextResponse:
    return PlainTextResponse("dalian")  # 返回纯文本响应


async def create_audio():
    return {
        "任务": "音频生成",
    }


async def modify_pic():
    return {
        "任务": "图像修改",
    }


async def test():
    return {
        "任务": "测试接口",
    }
